from .redis import AioRedisDatabaseProxy

_mapping = {'redis': AioRedisDatabaseProxy}


def get_db_proxy_cls(db_type):
    return _mapping[db_type]
